import os

from fastmcp import FastMCP,Context

#实例mcp
mcp = FastMCP("Common MCP Server")

@mcp.tool()
def add_tool(x:int,y:int,ctx:Context):
    """
    有两个数字相加的加法工具
    :param x: 第一个数字
    :param y: 第二个数字
    :return: 两个数字的和
    """
    request = ctx.get_http_request()
    print(" === headers:",request.headers)
    print("  获取到client端的token：",request.headers.get("authorization"))
    token = request.headers.get("authorization")
    if token != "Bearer 123456":
        raise ValueError("token错误,安全认证没有通过")
    return x+y

@mcp.tool()
def sub_tool(x:int,y:int):
    """
    有两个数字相减的减法工具
    :param x: 第一个数字
    :param y: 第二个数字
    :return: 两个数字的差
    """
    return x-y



if __name__ == "__main__":
    print(" MCP Server 启动中...")
    mcp.run(transport="sse")
